package com.ruoyi.framework.security.handle;

import java.io.IOException;
import java.io.Serializable;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.ruoyi.secureapi.cipher.CipherAlgorithmEnum;
import com.ruoyi.secureapi.cipher.CipherUtils;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.security.core.AuthenticationException;
import org.springframework.security.web.AuthenticationEntryPoint;
import org.springframework.stereotype.Component;
import com.alibaba.fastjson2.JSON;
import com.ruoyi.common.constant.HttpStatus;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.utils.ServletUtils;
import com.ruoyi.common.utils.StringUtils;

/**
 * 认证失败处理类 返回未授权
 * 
 * @author ruoyi
 */
@Component
public class AuthenticationEntryPointImpl implements AuthenticationEntryPoint, Serializable
{
    private static final long serialVersionUID = -8970718410437077606L;

    @Value("${secure-api.key}")
    private String key;

    @Value("${secure-api.iv}")
    private String iv;

    @Value("${secure-api.enabled}")
    private Boolean enabled;

    @Override
    public void commence(HttpServletRequest request, HttpServletResponse response, AuthenticationException e)
            throws IOException
    {
        int code = HttpStatus.UNAUTHORIZED;
        String msg = StringUtils.format("请求访问：{}，认证失败，无法访问系统资源", request.getRequestURI());
        String encrypt = null;
        if(enabled){
            CipherUtils cipherUtils = new CipherUtils(CipherAlgorithmEnum.AES_CBC_PKCS5);
            encrypt = cipherUtils.encrypt(JSON.toJSONString(AjaxResult.error(code, msg)), key, iv);
        }else {
            encrypt = JSON.toJSONString(AjaxResult.error(code, msg));
        }
        ServletUtils.renderString(response, encrypt);
    }
}
